import torch
import torchvision.models

# 方式一 加载模型
vgg16_model = torch.load("../vgg16/vgg16_false_method1.pth")
# print(vgg16_model)

# 方式二 加载模型
vgg16 = torchvision.models.vgg16(pretrained=False)
vgg16.load_state_dict(torch.load("../vgg16/vgg16_false_method2.pth"))
# vgg16_model = torch.load("vgg16_false_method2.pth")
# print(vgg16_model)